#!/bin/bash

SOC_VERSION=$1

atc --framework=5 --output=./mmpose/hrnet  --input_format=ND  --soc_version=${SOC_VERSION} \
    --model=./mmpose/hrnet.onnx \
    --input_shape="x:1,3,-1,-1" \
    --dynamic_dims="512,768;512,832;512,704;512,896;512,640;512,960;512,512;512,1152;512,576;512,1024;512,1088;512,1280;512,1984;512,1536;512,2048;512,2112;512,1216;512,1856;512,1344;512,1728;512,1920;768,512;704,512;576,512;832,512;640,512;960,512;896,512;1152,512;1088,512;1024,512;1344,512;512,1472;512,1600;512,1792;512,1408"
